//
//  UIView+Extension.swift
//  Weibo
//
//  Created by itteacher on 2017/7/22.
//  Copyright © 2017年 itteacher. All rights reserved.
//

import UIKit

//  截屏
func getScreenShot() -> UIImage {
    //  获取window
    let window = UIApplication.shared.keyWindow!
    //  1. 开启图片上下文
    UIGraphicsBeginImageContext(window.bounds.size)
    //  2. 截屏区域
    let rect = CGRect(origin: CGPoint.zero, size: window.bounds.size)
    //  afterScreenUpdates: 不需要更新后进行截屏，因为本身对图片质量要求不高
    //  drawHierarchy: 在指定区域内截取图片
    window.drawHierarchy(in: rect, afterScreenUpdates: false)
    //  3. 在图片上下文中获取图片
    let image = UIGraphicsGetImageFromCurrentImageContext()
    //  4. 结束图片上下文
    UIGraphicsEndImageContext()
    return image!
}
